package com.xm.sort_algorithm;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * Author：xm
 * Project：interview
 * Date：2025/8/10  0:31
 */

// 快速排序
public class _02_QuickSort {
    static List<Integer> count = new ArrayList<>();

    public static void main(String[] args) {
        // int[] arrTemp = {5, 1, 15, 30, 3, 15, 16};
        int[] arrTemp = {20, 2, 7, 12, 15, 1, 6, 8};
        System.out.println("排序前：" + Arrays.toString(arrTemp));
        Arrays.sort(arrTemp);
        System.out.println("排序后：" + Arrays.toString(arrTemp));

        // int[] arr = {5, 1, 15, 30, 3, 15, 16};
        // int[] arr = {50, 60, 70, 90, 80, 10, 20, 40};
        int[] arr = {20, 2, 7, 12, 15, 1, 6, 8};
        // int[] arr = {20, 2, 7, 12, 15, 1, 8, 8};
        quickSort(arr, 0, arr.length - 1);
        // Arrays.sort(arr);
        System.out.println("基准元素："+count);
        for (int i : arr) {
            System.out.print(i + " ");
        }
    }

    public static void quickSort(int[] arr, int left, int right) {
        // 递归终止条件
        if (left >= right) {
            return;
        }
        // 获取基准元素的位置
        int pivotIndex = partition(arr, left, right);
        System.out.println("分区：" + Arrays.toString(arr));
        // 递归排序左右两部分
        quickSort(arr, left, pivotIndex - 1);
        quickSort(arr, pivotIndex + 1, right);
    }

    public static int partition(int[] arr, int left, int right) {
        // 选择最右边的元素作为基准
        int pivot = arr[right];
        int i = left - 1;
        for (int j = left; j < right; j++) {
            if (arr[j] < pivot) {
                i++;
                // swap(arr, i, j);
                int temp = arr[i];
                arr[i] = arr[j];
                arr[j] = temp;
            }
        }

        // 将基准元素放到正确的位置
        // swap(arr, i + 1, right);
        int temp = arr[i + 1];
        arr[i + 1] = arr[right];
        arr[right] = temp;
        // 存储所有的基准元素。该行可删除，观察程序需要，无实际意义
        count.add(arr[i+1]);
        // 返回基准元素的索引
        return i + 1;
    }
    // 交换数组中的两个元素

    public static void swap(int[] arr, int i, int j) {
        int temp = arr[i];
        arr[i] = arr[j];
        arr[j] = temp;
    }
}
